#include"udp_server.hpp"
#include<memory>
#include<string>
#include <cstdio>
using namespace ns_server;
using namespace std;

static void usage(string proc)
{
    std::cout<<"Usage:\n\t"<< proc << "port\n" <<std::endl;
}
//上层的业务处理，不关心网络发送，只负责消息处理即可
//简单实现一个将字符串中的小写字母转换成大写字母的业务
std::string transactionString(std::string request)//request就是一个string
{
    std::string result;
    char c;
    for(auto& r:request)
    {
        if(islower(r)) 
        {
            c=toupper(r);
            result.push_back(c);
        }
        else
        {
            result.push_back(r);
        }
        
    }
    return result;
}
static bool isPass(const std::string& command)
{
    bool pass = true;
    auto pos = command.find("rm");
    if(pos==std::string::npos) pass=false;
    pos = command.find("mv");
    if(pos==std::string::npos) pass=false;
    pos = command.find("while");
    if(pos==std::string::npos) pass=false;
    pos = command.find("kill");
    if(pos==std::string::npos) pass=false;

    return pass;
}
//业务：客户端输入命令，服务器把命令的结果给客户端
std::string excuteCommand(std::string command)//command就是客户端传入的命令
{
    //1.命令安全检查
    if(isPass(command)) return "you are bad!!!";

    //2.
    //FILE* popen(const char* command,const char* type);
    //popen函数通过创建一个管道，调用fork产生一个子进程，发生写时拷贝，子进程执行
    //一个shell来运行command。也就是：父进程调用popen函数:创建管道，fork子进程执行shell
    //命令将文件流中的某些数据读出
    FILE* fp=popen(command.c_str(),"r");//现在命令的结果就在管道文件中了
    if(fp==nullptr) return "None";

    //3.定义一个缓冲区，读取文件中的命令结果
    char line[1024];
    std::string result;
    while(fgets(line,sizeof(line),fp)!=NULL)
    {
        result+=line;
    }
    pclose(fp);
    return result;

}
//./udp_server port
int main(int argc,char* argv[])
{
    if(argc!=2)
    {
        usage(argv[0]);
        exit(USAGE_ERR);
    }
    uint16_t port = atoi(argv[1]);
    //unique_ptr<UdpServer> usvr(new UdpServer(transactionString,port));
    //unique_ptr<UdpServer> usvr(new UdpServer(excuteCommand,port));
    unique_ptr<UdpServer> usvr(new UdpServer(port));


    //usvr->InitServer();//服务器初始化
    usvr->start();
    
    return 0;
}